#include "BlockQueue.hpp"
#include "Log.hpp"
#include "Task.hpp"
#include <string>
#include <time.h>

extern Log lg;

// void *Consumer(void *args)
// {
//     BlockQueue<int> *bq = static_cast<BlockQueue<int> *>(args);
//     while(true)
//     {
//         //消费
//         int data = bq->pop();
//         lg(Info, "消费了一个数据: %d, 剩余资源量: %d", data, bq->GetSize());
//         sleep(1);
//     }
// }

// void *Productor(void *args)
// {
//     BlockQueue<int> *bq = static_cast<BlockQueue<int> *>(args);
//     int data = 0;
//     while(true)
//     {
//         //生产
//         bq->push(++data);
//         lg(Info, "生产了一个数据: %d, 剩余资源量: %d", data, bq->GetSize());
//     }
// }

// int main()
// {
//     BlockQueue<int> *bq = new BlockQueue<int>();

//     pthread_t c,p;
//     pthread_create(&c, nullptr, Consumer, bq);
//     pthread_create(&p, nullptr, Productor, bq);
    
//     pthread_join(c, nullptr);
//     pthread_join(p, nullptr);

//     delete bq;

//     return 0;
// }

std::string opers = "+-*/%";

void *Productor(void *args)
{
    int len = opers.size();
    BlockQueue<Task> *bq = static_cast<BlockQueue<Task> *>(args);
    
    while(true)
    {
        //模拟生产者生产数据
        int data1 = rand() % 5;
        usleep(10);
        int data2 = rand() % 5; 
        char op = opers[rand() % len];
        Task t(data1, data2, op);

        //生产
        bq->push(t);
        lg(Info, "生产了一个任务, %s", t.GetTask().c_str());
        sleep(1);
    }
}

void *Consumer(void *args)
{
    BlockQueue<Task> *bq = static_cast<BlockQueue<Task> *>(args);
    while(true)
    {
        //消费
        Task t = bq->pop();
        
        //计算
        t.run();
        
        lg(Info, "处理任务: %s, 运算结果: %s", t.GetTask().c_str(), t.GetResult().c_str());
    }
}

int main()
{
    srand(time(nullptr));
    BlockQueue<Task> *bq = new BlockQueue<Task>();
    
    pthread_t c[3], p[5];
    for (int i = 0; i < 3; i++)
    {
        pthread_create(c + i, nullptr, Consumer, bq);
    }

    for (int i = 0; i < 5; i++)
    {
        pthread_create(p + i, nullptr, Productor, bq);
    }

    for (int i = 0; i < 3; i++)
    {
        pthread_join(c[i], nullptr);
    }
    for (int i = 0; i < 5; i++)
    {
        pthread_join(p[i], nullptr);
    }
    
    delete bq;
    return 0;
}
